from .loss import BaseLoss, CrossEntropyLoss, MSELoss